Python list comprehension invalid syntax